package com.ybwh.springboot2;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.ArrayList;
import java.util.List;

public class CodeGenerator2 {


    public static final String DB_NAME = "mysql01.dev.lls.com:5190/operation_client_demo";
    public static final String USER_NAME = "root";
    public static final String PASSWORD = "123456";

    //表明前缀,生成实体类名称会忽略这些前缀
    public static final String TABLE_PREFIX = "t_";


    //基本包名称
    public static final String BASE_PKG_NAME = "com.lls.operation.client.demo";

    //大项目路劲
    public static final String PROJECT_PATH = "D:/work/组件二组/运营中台/operation-client-demo";

    // service 项目 路径
    public static final String SERVICE_PATH = PROJECT_PATH + "/operation-client-demo-service";
    // service项目包路径 与 BASE_PKG_NAME 对应
    public static final String SERVICE_PKG_PATH = SERVICE_PATH + "/src/main/java/com/lls/operation/client/demo";
    // mapper的xml路径
    public static final String MAPPER_XML_PATH = SERVICE_PATH + "/src/main/resources/mappers/";


    //代码生成器 涉及表格 ，不指定表示所有表
    public static final String TABLES = "";

    // 生成代码注释中作者
    public static final String AUTHOR = "fanbeibei";

    public static void main(String[] args) {



        // 代码生成器
        AutoGenerator mpg = new AutoGenerator();

        GlobalConfig gc = new GlobalConfig();
        gc.setOutputDir("D:/tmp");
        gc.setFileOverride(true); // 默认 false ,是否覆盖已生成文件
        gc.setOpen(false); //默认true ,是否打开输出目录
        gc.setEnableCache(false); // 默认false,是否开启二级缓存
        gc.setAuthor(AUTHOR); // 作者
        gc.setSwagger2(true); //默认false
        gc.setBaseResultMap(true); // 默认false
        gc.setDateType(DateType.TIME_PACK); // 时间策略 默认TIME_PACK
        gc.setBaseColumnList(true); //默认false  和basemodel相似
        gc.setEntityName("%s");
        gc.setControllerName("%sController");
        gc.setServiceName("%sService");
        gc.setServiceImplName("%sServiceImpl");
        gc.setMapperName("%sMapper");
        gc.setIdType(IdType.AUTO); // 指定生成的主键类型
        mpg.setGlobalConfig(gc);


        // 数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl("jdbc:mysql://" + DB_NAME + "?useUnicode=true&useSSL=false&characterEncoding=utf8&serverTimezone=Asia/Shanghai");
        // dsc.setSchemaName("public");
        dsc.setDriverName("com.mysql.cj.jdbc.Driver");
        dsc.setUsername(USER_NAME);
        dsc.setPassword(PASSWORD);
        mpg.setDataSource(dsc);

        // 包配置
        PackageConfig pc = new PackageConfig();
//        pc.setModuleName(mouduleName);
        pc.setParent(BASE_PKG_NAME);
//        pc.setEntity("po");
        mpg.setPackageInfo(pc);

        // 自定义配置
        InjectionConfig cfg = new InjectionConfig() {
            @Override
            public void initMap() {
                // to do nothing


            }
        };


        // 自定义输出配置
        List<FileOutConfig> focList = new ArrayList<>();
        // 自定义配置会被优先输出

        focList.add(new FileOutConfig("mapper.xml.ftl") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
                return MAPPER_XML_PATH + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
            }
        });


        focList.add(new FileOutConfig("mapper.java.ftl") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
                return SERVICE_PKG_PATH + "/mapper/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_JAVA;
            }
        });

        focList.add(new FileOutConfig("entity.java.ftl") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
                return SERVICE_PKG_PATH + "/entity/" + tableInfo.getEntityName() + StringPool.DOT_JAVA;
            }
        });


        focList.add(new FileOutConfig("service.java.ftl") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
                return SERVICE_PKG_PATH + "/service/" + tableInfo.getEntityName() + "Service" + StringPool.DOT_JAVA;
            }
        });

        focList.add(new FileOutConfig("serviceImpl.java.ftl") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
                return  SERVICE_PKG_PATH + "/service/impl/" + tableInfo.getEntityName() + "ServiceImpl" + StringPool.DOT_JAVA;
            }
        });

        cfg.setFileOutConfigList(focList);

        mpg.setCfg(cfg);

        // 配置模板
        TemplateConfig templateConfig = new TemplateConfig();

        // 配置自定义输出模板
        //指定自定义模板路径，注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别
        templateConfig.setEntity("entity.java");
        templateConfig.setService("service.java");
        templateConfig.setServiceImpl("serviceImpl.java");
        templateConfig.setMapper("mapper.java");
        templateConfig.setXml("mapper.xml");
        //控制 不生成 controller  空字符串就行
        templateConfig.setController("");


//        sc.setControllerMappingHyphenStyle(false); //默认false


        mpg.setTemplate(templateConfig);

        // 策略配置
        StrategyConfig strategy = new StrategyConfig();

        strategy.setTablePrefix(TABLE_PREFIX);


        strategy.setCapitalMode(false); //是否大写命名 默认false
        strategy.setSkipView(true); //是否跳过试图 默认false
        strategy.setNaming(NamingStrategy.underline_to_camel);// 表映射 驼峰命名
        strategy.setColumnNaming(NamingStrategy.underline_to_camel); // 字段映射 驼峰
        strategy.setEntityLombokModel(true); //是否使用lombak 默认为false
        strategy.setRestControllerStyle(true); // 默认false
        strategy.setEntitySerialVersionUID(true); //默认true
        strategy.setEntityColumnConstant(true); //默认false 将mysql字段名生成静态变量
        if (null != TABLES && TABLES.trim().length() > 0) {
            strategy.setInclude(TABLES); //表名，用，隔开  需要生产
        }
        strategy.setEntityTableFieldAnnotationEnable(true); // 默认false 注释


        mpg.setStrategy(strategy);
        mpg.setTemplateEngine(new FreemarkerTemplateEngine());
        mpg.execute();

        System.out.println("-----------------CodeGenerator  compelete !!--------------------");
    }

}